FROM artifacts.lenovo.com/cybertron/platform/cann:8.2.rc1-910b-ubuntu22.04-py3.11


RUN apt-get update && \
    apt-get install -y vim && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*


RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip config set global.extra-index-url https://download.pytorch.org/whl/cpu && \
    pip install torch-npu==2.6.0 jupyterlab

WORKDIR /workspace

EXPOSE 8888

CMD ["jupyter-lab", "--ip=0.0.0.0", "--port=8888", "--allow-root", "--no-browser"]